stage=0


# train_data="audio_data/train_20250422_llmasr_3kh.jsonl,audio_data/train_20231206_slidespeech_487h.jsonl,audio_data/train_20250417_cn_slides_generated_context_all_upper.jsonl,audio_data/train_20250410_cn_slides_draft_8kh_open_ocr_all_upper.jsonl"
#train_data="audio_data/train_20231206_slidespeech_487h.jsonl"
# train_data="audio_data/train_ver250425/train_20231206_slidespeech_487h.jsonl,audio_data/train_ver250425/train_20250425_cn_slides_generated_context_neg3w.jsonl,audio_data/train_ver250425/train_20250425_llm_asr_3kh_remove_repeat.jsonl,audio_data/train_ver250425/train_20250410_cn_slides_draft_8kh_open_ocr_all_upper.jsonl"
#valid_data="audio_data/test_20220221_ailab_5h_org_fbankhires.jsonl"

train_data="audio_data/train_ver250509/train_20231206_slidespeech_487h.jsonl,audio_data/train_ver250509/train_20250425_llm_asr_3kh_remove_repeat.jsonl,audio_data/train_ver250509/train_20250325_cn_slides_draft_8kh.jsonl.upper,audio_data/train_ver250509/train_20250410_cn_slides_draft_8kh_open_ocr_all_upper.jsonl,audio_data/train_ver250509/train_20250425_cn_slides_generated_context_neg3w.jsonl,audio_data/train_ver250509/train_20250425_llm_asr_3kh_remove_repeat.rep2.jsonl,audio_data/train_ver250509/train_20250425_llm_asr_3kh_remove_repeat.rep3.jsonl"



valid_data="audio_data/valid_20240322_cn_slides_1100h.jsonl,audio_data/test_20241119_slidespeech.jsonl,audio_data/context_test/yuanbao_pst_sf.jsonl,audio_data/context_test/yuanbao_pst_test_long_exact_kw_wemeet.jsonl"
#train_data=$valid_data

#train_data="audio_data/train_20231206_slidespeech_487h.jsonl"
#valid_data="audio_data/test_20241119_slidespeech.jsonl"

#train_data=audio_data/train_aishell1.jsonl
#valid_data=audio_data/dev_aishell1.jsonl
#train_data="audio_data/train_sub_asr_zhen_train_20240322_cn_slides_asr_multi.jsonl"
#valid_data="audio_data/valid_20240322_cn_slides_1100h_asr_multi.jsonl"
#train_data=$valid_data
#save_prefix=audiollm_asr_whisper_tiny_qwen0.5b_zhen_use_specaug
#save_prefix=audiollm_asr_llmasr_3kh_context_zhen_use_specaug_250324
save_prefix=moeconformerllm_asr_llmasr_zhen_slides_specaug_audio_last_250510

. utils/parse_options.sh







if [ $stage -le 0 ]; then
    echo "Stage 0: Training Model"
#    bash launch_distributed_password.sh \
    bash launch_distributed.sh \
        --DATA_PATH $train_data \
	--VALID_DATA_PATH $valid_data \
	--SAVE_PREFIX $save_prefix \
        --SCRIPT scripts/run_moeconformerllm.sh
	#--SCRIPT scripts/train_shallow_audiollm.sh
fi

if [ $stage -le 1 ]; then
    echo "Stage 1: Run EMPTY"
    bash launch_distributed.sh \
	 --SAVE_PREFIX empty_debug \
	 --SCRIPT scripts/empty.sh

fi
